/***
题目链接：堆盘子
堆盘子。设想有一堆盘子，堆太高可能会倒下来。因此，在现实生活中，盘子堆到一定高度时，我们就会另外堆一堆盘子。请实现数据结构SetOfStacks，模拟这种行为。SetOfStacks应该由多个栈组成，并且在前一个栈填满时新建一个栈。此外，SetOfStacks.push()和SetOfStacks.pop()应该与普通栈的操作方法相同（也就是说，pop()返回的值，应该跟只有一个栈时的情况一样）。 
进阶：实现一个popAt(int index)方法，根据指定的子栈，执行pop操作。
当某个栈为空时，应当删除该栈。当栈中没有元素或不存在该栈时，pop，popAt 应返回 -1.
链接：https://leetcode-cn.com/problems/stack-of-plates-lcci
***/
class StackOfPlates {
private:
    int capacity;
    vector <stack<int>> vec;
public:
    StackOfPlates(int cap) {
        capacity = cap;
    }
    
    void push(int val) {
        //判断要是满栈就新建一个栈，然后加进去
        //要是空栈也加入一个栈
        if (vec.empty() || vec.back().size()==capacity){
            vec.push_back(stack<int>());
        }
        //要是没有满栈就加到最后一个栈当中
        if(vec.back().size() < capacity) vec.back().push(val);
    }
    
    int pop() {
        return popAt(vec.size()-1);
    }
    
    int popAt(int index) {
        int stack_top = -1;
        if (index < vec.size()){
            //记录栈顶元素，并且弹出栈顶元素，
            //弹出之后判断该栈是否空栈了
            if(!vec[index].empty()){
                stack_top = vec[index].top();
                vec[index].pop();
            }
            if (vec[index].empty()) vec.erase(vec.begin()+index);
        }
        return stack_top;
    }
};

/**
 * Your StackOfPlates object will be instantiated and called as such:
 * StackOfPlates* obj = new StackOfPlates(cap);
 * obj->push(val);
 * int param_2 = obj->pop();
 * int param_3 = obj->popAt(index);
 */